"""
Make sure we generate a manifest file when linking binaries, including
handling AdditionalManifestFiles.
"""

import TestGyp

test = TestGyp.TestGyp(formats=['msvs', 'ninja'], platform=['win32'], disable='Need to solve win32api.LoadLibrary problems')

import pywintypes
import win32api
import winerror

RT_MANIFEST = 24

class LoadLibrary(object):
  """Context manager for loading and releasing binaries in Windows.
  Yields the handle of the binary loaded."""

  def __init__(self, path):
    self._path = path
    self._handle = None

  def __enter__(self):
    self._handle = win32api.LoadLibrary(self._path)
    return self._handle

  def __exit__(self, type, value, traceback):
    win32api.FreeLibrary(self._handle)


def extract_manifest(path, resource_name):
  """Reads manifest from |path| and returns it as a string.
  Returns None is there is no such manifest."""
  with LoadLibrary(path) as handle:
    try:
      return win32api.LoadResource(handle, RT_MANIFEST, resource_name)
    except pywintypes.error as error:
      if error.args[0] == winerror.ERROR_RESOURCE_DATA_NOT_FOUND:
        return None
      else:
        raise


CHDIR = 'linker-flags'
test.run_gyp('generate-manifest.gyp', chdir=CHDIR)
test.build('generate-manifest.gyp', test.ALL, chdir=CHDIR)

# Make sure that generation of .generated.manifest does not cause a relink.
test.run_gyp('generate-manifest.gyp', chdir=CHDIR)
test.up_to_date('generate-manifest.gyp', test.ALL, chdir=CHDIR)


def test_manifest(filename, generate_manifest, embedded_manifest, extra_manifest):
  exe_file = test.built_file_path(filename, chdir=CHDIR)
  if not generate_manifest:
    test.must_not_exist(exe_file + '.manifest')
    manifest = extract_manifest(exe_file, 1)
    test.fail_test(manifest)
    return
  if embedded_manifest:
    manifest = extract_manifest(exe_file, 1)
    test.fail_test(not manifest)
  else:
    test.must_exist(exe_file + '.manifest')
    manifest = test.read(exe_file + '.manifest')
    test.fail_test(not manifest)
    test.fail_test(extract_manifest(exe_file, 1))
  if generate_manifest:
    test.must_contain_any_line(manifest, 'requestedExecutionLevel')
  if extra_manifest:
    test.must_contain_any_line(manifest, '35138b9a-5d96-4fbd-8e2d-a2440225f93a')
    test.must_contain_any_line(manifest, 'e2011457-1546-43c5-a5fe-008deee3d3f0')


test_manifest('test_generate_manifest_true.exe', generate_manifest=True, embedded_manifest=False, extra_manifest=False)
test_manifest('test_generate_manifest_false.exe', generate_manifest=False, embedded_manifest=False, extra_manifest=False)
test_manifest('test_generate_manifest_default.exe', generate_manifest=True, embedded_manifest=False, extra_manifest=False)
test_manifest('test_generate_manifest_true_as_embedded.exe', generate_manifest=True, embedded_manifest=True, extra_manifest=False)
test_manifest('test_generate_manifest_false_as_embedded.exe', generate_manifest=False, embedded_manifest=True, extra_manifest=False)
test_manifest('test_generate_manifest_default_as_embedded.exe', generate_manifest=True, embedded_manifest=True, extra_manifest=False)
test_manifest('test_generate_manifest_true_with_extra_manifest.exe', generate_manifest=True, embedded_manifest=False, extra_manifest=True)
test_manifest('test_generate_manifest_false_with_extra_manifest.exe', generate_manifest=False, embedded_manifest=False, extra_manifest=True)
test_manifest('test_generate_manifest_true_with_extra_manifest_list.exe', generate_manifest=True, embedded_manifest=False, extra_manifest=True)
test_manifest('test_generate_manifest_false_with_extra_manifest_list.exe', generate_manifest=False, embedded_manifest=False, extra_manifest=True)
test_manifest('test_generate_manifest_default_embed_default.exe', generate_manifest=True, embedded_manifest=True, extra_manifest=False)
test.pass_test()
